package com.xiaoniu.forum.common.utils;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import org.springframework.util.StringUtils;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.function.Consumer;
import java.util.function.Function;

/**
 * 构造查询条件的工具类
 * 根据一定格式的json字符串，构建出查询QueryWrapper （只适用于单表查询）
 */
public class QueryWrapperUtils {
    public static <T> QueryWrapper<T> createQueryWrapper(CommonQueryVo queryVo,Class<T> clazz){
        QueryWrapper<T> queryWrapper=new QueryWrapper();
        if (queryVo.getAnd()!=null){
            buildAnd(queryVo,queryWrapper);
        }
        if (queryVo.getOr()!=null){
            buildOr(queryVo,queryWrapper);
        }
        return queryWrapper;
    }

    /**
     * 构建and条件到QueryWrapper对象中
     */
    private static void buildOr(CommonQueryVo queryVo,QueryWrapper queryWrapper){
        CommonQueryVOItem voItem = queryVo.getOr();
        addCriteria(voItem.getGe(),queryWrapper,(wrapper,key,value)->{
            wrapper.or();
            wrapper.ge(key,value);
        });
        addCriteria(voItem.getGt(),queryWrapper,(wrapper,key,value)->{
            wrapper.or();
            wrapper.gt(key,value);
        });
        addCriteria(voItem.getLe(),queryWrapper,(wrapper,key,value)->{
            wrapper.or();
            wrapper.le(key,value);
        });
        addCriteria(voItem.getLt(),queryWrapper,(wrapper,key,value)->{
            wrapper.or();
            wrapper.lt(key,value);
        });
        addCriteria(voItem.getMust(),queryWrapper,(wrapper,key,value)->{
            wrapper.or();
            wrapper.eq(key,value);
        });
        addCriteria(voItem.getMustNot(),queryWrapper,(wrapper,key,value)->{
            wrapper.or();
            wrapper.ne(key,value);
        });
        addCriteria(voItem.getLike(),queryWrapper,(wrapper,key,value)->{
            wrapper.or();
            wrapper.like(key,value);
        });
        if (voItem.getBetween()!=null){
            for (String[] ss:voItem.getBetween()){
                if (!StringUtils.isEmpty(ss[0]) && !StringUtils.isEmpty(ss[1]) && !StringUtils.isEmpty(ss[2])){
                    queryWrapper.or();
                    queryWrapper.between(ss[0],ss[1],ss[2]);
                }
            }
        }
    }


    /**
     * 构建or条件到QueryWrapper对象中
     */
    private static void buildAnd(CommonQueryVo queryVo,QueryWrapper queryWrapper){
        CommonQueryVOItem voItem = queryVo.getAnd();
        addCriteria(voItem.getGe(),queryWrapper,(wrapper,key,value)->{
            wrapper.ge(key,value);
        });
        addCriteria(voItem.getGt(),queryWrapper,(wrapper,key,value)->{
            wrapper.gt(key,value);
        });
        addCriteria(voItem.getLe(),queryWrapper,(wrapper,key,value)->{
            wrapper.le(key,value);
        });
        addCriteria(voItem.getLt(),queryWrapper,(wrapper,key,value)->{
            wrapper.lt(key,value);
        });
        addCriteria(voItem.getMust(),queryWrapper,(wrapper,key,value)->{
            wrapper.eq(key,value);
        });
        addCriteria(voItem.getMustNot(),queryWrapper,(wrapper,key,value)->{
            wrapper.ne(key,value);
        });
        addCriteria(voItem.getLike(),queryWrapper,(wrapper,key,value)->{
            wrapper.like(key,value);
        });
        if (voItem.getBetween()!=null){
            for (String[] ss:voItem.getBetween()){
                if (ss.length!=3) continue;
                if (!StringUtils.isEmpty(ss[0]) && !StringUtils.isEmpty(ss[1]) && !StringUtils.isEmpty(ss[2])){
                    queryWrapper.between(ss[0],ss[1],ss[2]);
                }
            }
        }

    }

    //添加查询条件到QueryWrapper中
    private static void addCriteria(Map<String,String> map,QueryWrapper queryWrapper,MyConsumer<QueryWrapper,String,Object> consumer){
        if(map!=null){
            for (String key:map.keySet()){
                String value = map.get(key);
                if (!StringUtils.isEmpty(value)){
                    consumer.accept(queryWrapper,key,value);
                }
            }
        }
    }
}

interface MyConsumer<Q,K,V>{
    void accept(Q q,K k,V v);
}
